import qiniu


# 需要填写你的 Access Key 和 Secret Key
access_key = "W0oGRaBkAhrcppAbz6Nc8-q5EcXfL5vLRashY4SI"
secret_key = "tsYCBckepW4CqW0uHb9RdfDMXRDOTEpYecJAMItL"
# 上传的图片的空间名称
bucket_name = "information22"


def storage(data):
    """七牛云储存上传文件接口"""
    if not data:
        return None

    # 创建权限对象
    q = qiniu.Auth(access_key, secret_key)
    # 上传的图片名称，可以不用设置，七牛云自动生成
    # key = 'hello'

    token = q.upload_token(bucket_name)
    # 上传文件
    ret, info = qiniu.put_data(token, None, data)

    if ret is not None:
        print("All is Ok")
    else:
        raise AttributeError("上传文件到七牛失败")

    if info.status_code == 200:
        print("上传图片成功")
    else:
        raise AttributeError("上传文件到七牛失败")

    # 返回图片名称
    return ret["key"]


if __name__ == '__main__':
    file_name = input("输入上传的文件")
    with open(file_name, "rb") as f:
        storage(f.read())
